// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <stddef.h>
#include <stdint.h>

#include <memory>

#include "base/macros.h"
#include "base/threading/thread_task_runner_handle.h"
#include "content/child/indexed_db/mock_webidbcallbacks.h"
#include "content/child/indexed_db/webidbdatabase_impl.h"
#include "content/child/thread_safe_sender.h"
#include "content/common/indexed_db/indexed_db_key.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/WebKit/public/platform/WebBlobInfo.h"
#include "third_party/WebKit/public/platform/WebData.h"
#include "third_party/WebKit/public/web/WebHeap.h"

using blink::WebBlobInfo;
using blink::WebData;
using blink::WebIDBCursor;
using blink::WebIDBKey;
using blink::WebVector;
using testing::_;
using testing::Invoke;
using testing::StrictMock;
using testing::WithArgs;

namespace content {

class WebIDBDatabaseImplTest : public testing::Test {
public:
    WebIDBDatabaseImplTest() { }

    void TearDown() override { blink::WebHeap::collectAllGarbageForTesting(); }

private:
    TestBrowserThreadBundle thread_bundle_;

    DISALLOW_COPY_AND_ASSIGN(WebIDBDatabaseImplTest);
};

TEST_F(WebIDBDatabaseImplTest, ValueSizeTest)
{
    // For testing use a much smaller maximum size to prevent allocating >100 MB
    // of memory, which crashes on memory-constrained systems.
    const size_t kMaxValueSizeForTesting = 10 * 1024 * 1024; // 10 MB

    const std::vector<char> data(kMaxValueSizeForTesting + 1);
    const WebData value(&data.front(), data.size());
    const WebVector<WebBlobInfo> web_blob_info;

    const int64_t transaction_id = 1;
    const int64_t object_store_id = 2;

    StrictMock<MockWebIDBCallbacks> callbacks;
    EXPECT_CALL(callbacks, onError(_)).Times(1);

    WebIDBDatabaseImpl database_impl(nullptr,
        base::ThreadTaskRunnerHandle::Get());
    database_impl.max_put_value_size_ = kMaxValueSizeForTesting;
    database_impl.put(transaction_id, object_store_id, value, web_blob_info,
        WebIDBKey::createNumber(0), blink::WebIDBPutModeAddOrUpdate,
        &callbacks, WebVector<long long>(),
        WebVector<WebVector<WebIDBKey>>());
}

TEST_F(WebIDBDatabaseImplTest, KeyAndValueSizeTest)
{
    // For testing use a much smaller maximum size to prevent allocating >100 MB
    // of memory, which crashes on memory-constrained systems.
    const size_t kMaxValueSizeForTesting = 10 * 1024 * 1024; // 10 MB
    const size_t kKeySize = 1024 * 1024;

    const std::vector<char> data(kMaxValueSizeForTesting - kKeySize);
    const WebData value(&data.front(), data.size());
    const WebVector<WebBlobInfo> web_blob_info;
    const WebIDBKey key = WebIDBKey::createString(
        base::string16(kKeySize / sizeof(base::string16::value_type), 'x'));

    const int64_t transaction_id = 1;
    const int64_t object_store_id = 2;

    StrictMock<MockWebIDBCallbacks> callbacks;
    EXPECT_CALL(callbacks, onError(_)).Times(1);

    WebIDBDatabaseImpl database_impl(nullptr,
        base::ThreadTaskRunnerHandle::Get());
    database_impl.max_put_value_size_ = kMaxValueSizeForTesting;
    database_impl.put(transaction_id, object_store_id, value, web_blob_info, key,
        blink::WebIDBPutModeAddOrUpdate, &callbacks,
        WebVector<long long>(), WebVector<WebVector<WebIDBKey>>());
}

} // namespace content
